hackerrank list comprehension solution